# import os
# import wfdb
# import psycopg2
# from psycopg2 import sql
# import json
# import numpy as np
# # Function to fetch data for one user
# def fetch_user_data(user_id, wfdb_records_dir):
# try:
# # Construct the paths to .hea and .mat files
# header_file = os.path.join(wfdb_records_dir, user_id + '.hea')
# data_file = os.path.join(wfdb_records_dir, user_id + '.mat')
# # Check if both .hea and .mat files exist
# if os.path.exists(header_file) and os.path.exists(data_file):
# # Load ECG data and metadata
# record = wfdb.rdrecord(os.path.join(wfdb_records_dir, user_id))
# ecg_data = record.p_signal
# metadata = record.comments
# return ecg_data, metadata
# else:
# print(f"Data files not found for user {user_id}")
# return None, None
# except Exception as e:
# print(f"Error fetching data for user {user_id}: {e}")
# return None, None
# # Function to replace "NaN" values with None in the nested list
# def replace_nan_with_none(data):
# if isinstance(data, list):
# return [replace_nan_with_none(x) for x in data]
# elif isinstance(data, float) and np.isnan(data):
# return None
# else:
# return data
# # Function to traverse directories recursively and fetch data
# def traverse_directories(wfdb_records_base_dir):
# for root, dirs, files in os.walk(wfdb_records_base_dir):
# for dir in dirs:
# records_dir = os.path.join(root, dir)
# records_file = os.path.join(records_dir, 'RECORDS')
# if os.path.exists(records_file):
# with open(records_file, 'r') as f:
# for line in f:
# user_id = line.strip()
# ecg_data, metadata = fetch_user_data(user_id, records_dir)
# if ecg_data is not None and metadata is not None:
# # Replace "NaN" values with None
# ecg_data_cleaned = replace_nan_with_none(ecg_data.tolist())
# ecg_data_json = json.dumps(ecg_data_cleaned)
# if isinstance(metadata, list):
# metadata_json = json.dumps(metadata)
# else:
# metadata_json = metadata
# cur.execute(sql.SQL("""
# INSERT INTO WFDBRecords (user_id, ecg_data, metadata)
# VALUES (%s, %s, %s)
# """), (user_id, ecg_data_json, metadata_json))
# conn.commit()
# print(f"User {user_id} added to the database. Path: {records_dir}")
# else:
# print(f"Skipping user {user_id} due to missing data in {records_dir}")
# # Directory where WFDB records are stored locally
# wfdb_records_base_dir = 'D:/New/Data Science/Assignment_3/a-large-scale-12-lead-electrocardiogram-database-for-arrhythmia-study-1.0.0/WFDBRecords'
# # Connect to the PostgreSQL database
# try:
# conn = psycopg2.connect(
# dbname="postgres",
# user="postgres",
# password="dost",
# host="localhost",
# port="1234"
# )
# cur = conn.cursor()
# # Create a table to store ECG data and metadata if not exists
# cur.execute("""
# CREATE TABLE IF NOT EXISTS WFDBRecords (
# id SERIAL PRIMARY KEY,
# user_id VARCHAR,
# ecg_data JSONB,
# metadata JSONB
# );
# """)
# conn.commit()
# # Traverse directories recursively and fetch data
# traverse_directories(wfdb_records_base_dir)
# except psycopg2.Error as e:
# print("Error connecting to PostgreSQL:", e)
# finally:
# # Close the cursor and connection
# if 'cur' in locals():
# cur.close()
# if 'conn' in locals():
# conn.close()
